使用免费的 CodeGemma

之前介绍过Google的开放模型Gemma（Gemma - 谷歌最新开放模型）。今天看看他的CodeGemma，据说比CodeLlama好用。（CodeLlama我之前也分享过Code Llama AI 代码生成）

Colab代码详见：https://colab.research.google.com/github/google/generative-ai-docs/blob/main/site/en/gemma/docs/codegemma/keras_quickstart.ipynb

运行前需要配置以下环境：

1. 在Kaggle的设置里面创建新的API令牌（API create new token）。
2. 在Colab中，选择“Secrets”（🔑），然后添加你的Kaggle用户名和Kaggle API密钥。将你的用户名存储在名称为KAGGLE_USERNAME的变量中，API密钥存储在名称为KAGGLE_KEY的变量中。

此外，还需要在Kaggle的模型卡中签署下载CodeGemma的协议。

完成以上配置后，你就可以运行CodeGemma了。

例如，我让CodeGemma给我写一个2048游戏的代码。

generation_prompt= """Write 2048 game"""
print(gemma_lm.generate(generation_prompt, max_length=5000))